package com.hina.rec.endpoint.res.base;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;

/**
 * 分页响应对象
 *
 * @author liaosp
 * @date 2023-06-01
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class PageRes<T> {
    /**
     * 总记录数
     */
    private long total;

    /**
     * 总页数
     */
    private long pages;

    /**
     * 当前页码
     */
    private long pageNum;

    /**
     * 每页数量
     */
    private long pageSize;

    /**
     * 数据列表
     */
    private List<T> list;


    /**
     * 将 MyBatis-Plus 的 Page 对象转换为 PageRes 对象
     *
     * @param page MyBatis-Plus 的分页对象
     * @param <T>  数据类型
     * @return 转换后的 PageRes 对象
     */
    public static <T> PageRes<T> fromPage(Page<T> page) {
        PageRes<T> pageRes = new PageRes<>();
        pageRes.setTotal(page.getTotal());
        pageRes.setPages(page.getPages());
        pageRes.setPageNum(page.getCurrent());
        pageRes.setPageSize(page.getSize());
        pageRes.setList(page.getRecords());
        return pageRes;
    }

    /**
     * 如果需要对数据进行额外处理（如 DTO 转换），可以使用此方法
     *
     * @param page      MyBatis-Plus 的分页对象
     * @param converter 数据转换函数
     * @param <S>       原始数据类型
     * @param <T>       目标数据类型
     * @return 转换后的 PageRes 对象
     */
    public static <S, T> PageRes<T> fromPage(Page<S> page, java.util.function.Function<S, T> converter) {
        PageRes<T> pageRes = new PageRes<>();
        pageRes.setTotal(page.getTotal());
        pageRes.setPages(page.getPages());
        pageRes.setPageNum(page.getCurrent());
        pageRes.setPageSize(page.getSize());
        pageRes.setList(page.getRecords().stream().map(converter).toList());
        return pageRes;
    }
}